import torch

# Model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)

# Images
imgs = ['demo_new.png']  # batch of images

# Inference
results = model(imgs)

# Results
results.print()

print(f"{results=}")

print(results.xyxy[0])
